pytorch neural network from scratch